# -*- coding: utf-8 -*-
# Version: Python 3.9.7
# Author: TRIX
# Date: 2021-09-11 12:50:29
# Use:批量给图片添加水印图片 可选方位
from PIL import Image,ImageFont,ImageDraw
from os import listdir,path,mkdir
def add_watermark(folder,watermark,position):
	"""选择文件夹 使用前保证文件夹里全是图片文件 可选方位 左上-lt 左下-lb 右上-rt 右下-rb 正中-c 正下-cb 正上-ct"""
	wmPic=Image.open(watermark)
	for file in listdir(folder):
		pic=Image.open(folder+'\\'+file)
		picW,picH=pic.size

		if picW >= picH:#判断水印和图片的大小关系
			resW=int(picW*0.1)
			resH=int(picW*0.1)
		elif picW < picH:
			resW=int(picH*0.1)
			resH=int(picH*0.1)
		wmResize=wmPic.resize((resW,resH))

		if position=='rb':#判断水印和图片的位置关系
			coord=(picW-resW,picH-resH)
		elif position=='cb':
			coord=(int((picW-resW)/2),picH-resH)
		elif position=='ct':
			coord=(int((picW-resW)/2),0)
		elif position=='c':
			coord=(int((picW-resW)/2),int((picW-resW)/2))
		elif position=='lb':
			coord=(0,picH-resH)
		elif position=='rt':
			coord=(picW-resW,0)
		elif position=='lt':
			coord=(0,0)

		pic.paste(wmResize,coord,wmResize)#坐标是左上角 因为水印是透明的 所以有最后一个参数 直接在原图上粘贴
		if not path.exists(folder+'_wmed'):
			mkdir(folder+'_wmed')#如果输出文件夹不存在 创建文件夹
		pic.save(folder+'_wmed\\'+file)

add_watermark(r'.\hospital',r'.\watermark.png','rb')
